from enum import IntEnum

from pydantic import BaseModel, Field


class _IE(IntEnum):
    NEGATIVE_ONE = -1
    ZERO = 0
    POSITIVE_ONE = 1


class FakeCoinResultSchema(BaseModel):
    """
    fake_coin_puzzle_get_result_route 参数
    """

    matrix: list[list[_IE]] = Field(..., description="硬币分布矩阵")
    result_list: list[_IE] = Field(..., description="硬币分布结果")
